package com.example.demomongodb.utils;

import java.util.ArrayList;
import java.util.List;

import org.bson.Document;
import org.bson.conversions.Bson;
import org.bson.types.ObjectId;

import com.mongodb.BasicDBObject;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientOptions;
import com.mongodb.MongoClientOptions.Builder;
import com.mongodb.WriteConcern;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.MongoIterable;
import com.mongodb.client.model.Filters;
import com.mongodb.client.result.DeleteResult;

/**
 * MongoDB工具类 Mongo实例代表了一个数据库连接池，即使在多线程的环境中，一个Mongo实例对我们来说已经足够了
 * 注意Mongo已经实现了连接池，并且是线程安全的
 * 设计为单例模式， 因 MongoDB的Java驱动是线程安全的，对于一般的应用，只要一个Mongo实例即可
 * Mongo有个内置的连接池（默认为10个） 对于有大量写和读的环境中，为了确保在一个Session中使用同一个DB时
 * DB和DBCollection是绝对线程安全的
 *
 * @author Lewis
 * @date 2015-5-29 上午11:49:49
 */
public enum MongoDBUtil {

	/**
	 * 定义一个枚举的元素，它代表此类的一个实例
	 */
	instance;

	private MongoClient mongoClient;

	/**
	 * 地址
	 */
	private final String host = "localhost";

	/**
	 * 端口
	 */
	private final int port = 27017;

	static {
		System.out.println("===============MongoDBUtil初始化========================");
		// 单个连接
		instance.mongoClient = new MongoClient(instance.host, instance.port);

		// 或者，为了连接到一个复制集，通过自动发现主节点，提供成员的种子列表
		// List<ServerAddress> listHost = Arrays.asList(new ServerAddress("localhost", 27017),new ServerAddress("localhost", 27018));
		// instance.mongoClient = new MongoClient(listHost);

		// 大部分用户使用mongodb都在安全内网下，但如果将mongodb设为安全验证模式，就需要在客户端提供用户名和密码：
		// boolean auth = db.authenticate(myUserName, myPassword);

		Builder options = new MongoClientOptions.Builder();
		// 连接池设置为300个连接,默认为100
		options.connectionsPerHost(300);
		// 连接超时，推荐>3000毫秒
		options.connectTimeout(15000);
		options.maxWaitTime(5000);
		// 套接字超时时间，0无限制
		options.socketTimeout(0);
		// 线程队列数，如果连接线程排满了队列就会抛出“Out of semaphores to get db”错误。
		options.threadsAllowedToBlockForConnectionMultiplier(5000);
		options.writeConcern(WriteConcern.SAFE);
		options.build();
	}

	/**
	 * 获取DB实例 - 指定DB
	 *
	 * @param dbName
	 * @return
	 */
	public MongoDatabase getDb(String dbName) {
		if (dbName != null && !"".equals(dbName)) {
			return mongoClient.getDatabase(dbName);
		}
		return null;
	}

	/**
	 * 获取collection对象 - 指定Collection
	 *
	 * @param collName
	 * @return
	 */
	public MongoCollection<Document> getTableInDbName(String dbName, String collName) {
		if (null == collName || "".equals(collName)) {
			return null;
		}
		if (null == dbName || "".equals(dbName)) {
			return null;
		}
		return mongoClient.getDatabase(dbName).getCollection(collName);
	}

	/**
	 * 查询DB下的所有表名
	 */
	public List<String> getAllTable(String dbName) {
		MongoIterable<String> colls = getDb(dbName).listCollectionNames();
		List<String> list = new ArrayList<>();
		for (String s : colls) {
			list.add(s);
		}
		return list;
	}

	/**
	 * 获取所有数据库名称列表
	 *
	 * @return
	 */
	public MongoIterable<String> getAllDbNames() {
		return mongoClient.listDatabaseNames();
	}

	/**
	 * 删除一个数据库
	 */
	public void dropDb(String dbName) {
		getDb(dbName).drop();
	}

	/**
	 * 查找对象 - 根据主键_id
	 *
	 * @param coll
	 * @param id
	 * @return Document
	 */
	public Document getById(MongoCollection<Document> coll, String id) {
		ObjectId _id = null;
		try {
			_id = new ObjectId(id);
		} catch (Exception e) {
			return null;
		}
		return coll.find(Filters.eq("_id", _id)).first();
	}

	/**
	 * 统计数
	 */
	public int getCount(MongoCollection<Document> coll) {
		return (int) coll.count();
	}

	/**
	 * 条件查询
	 */
	public MongoCursor<Document> select(MongoCollection<Document> coll, Bson filter) {
		return coll.find(filter).iterator();
	}

	/**
	 * 分页查询
	 */
	public MongoCursor<Document> page(MongoCollection<Document> coll, Bson filter, int pageNo, int pageSize) {
		Bson orderBy = new BasicDBObject("_id", 1);
		return coll.find(filter).sort(orderBy).skip((pageNo - 1) * pageSize).limit(pageSize).iterator();
	}

	/**
	 * 通过ID删除
	 *
	 * @param coll
	 * @param id
	 * @return int
	 */
	public int deleteById(MongoCollection<Document> coll, String id) {
		int count = 0;
		ObjectId _id = null;
		try {
			_id = new ObjectId(id);
		} catch (Exception e) {
			return 0;
		}
		Bson filter = Filters.eq("_id", _id);
		DeleteResult deleteResult = coll.deleteOne(filter);
		count = (int) deleteResult.getDeletedCount();
		return count;
	}

	/**
	 * 通过ID修改
	 *
	 * @param coll
	 * @param id
	 * @param newdoc
	 * @return
	 */
	public Document updateById(MongoCollection<Document> coll, String id, Document newdoc) {
		ObjectId _idobj = null;
		try {
			_idobj = new ObjectId(id);
		} catch (Exception e) {
			return null;
		}
		Bson filter = Filters.eq("_id", _idobj);
		// 完全替代
		// coll.replaceOne(filter, newdoc);
		coll.updateOne(filter, new Document("$set", newdoc));
		return newdoc;
	}

	/**
	 * 删除指定库指定表
	 *
	 * @param dbName
	 * @param collName
	 */
	public void deleteTable(String dbName, String collName) {
		getDb(dbName).getCollection(collName).drop();
	}

	/**
	 * 插入单条数据
	 * @param dbName
	 * @param collName
	 * @param doc
	 */
	public void insertOne(String dbName, String collName, Document doc) {
		MongoCollection<Document> coll = MongoDBUtil.instance.getTableInDbName(dbName, collName);
		coll.insertOne(doc);
	}

	/**
	 * 插入复数条数据
	 * @param dbName
	 * @param collName
	 * @param docs
	 */
	public void insertAll(String dbName, String collName, List<Document> docs) {
		for (Document doc : docs) {
			insertOne(dbName, collName, doc);
		}
	}

	/**
	 * 关闭Mongodb
	 */
	public void close() {
		if (mongoClient != null) {
			mongoClient.close();
			mongoClient = null;
		}
	}

	public static void main(String[] args) {

		MongoIterable<String> allDBNames = instance.getAllDbNames();
		for (String dbName : allDBNames) {
			System.out.println("所有数据库 --> " + dbName);
		}

		String dbName = "lewis-demo-mongodb";
		String collName = "user";
		MongoCollection<Document> coll = MongoDBUtil.instance.getTableInDbName(dbName, collName);

		// 插入多条
		for (int i = 1; i <= 4; i++) {
			Document doc = new Document();
			doc.put("name", "zhoulf");
			doc.put("school", "NEFU" + i);
			instance.insertOne(dbName, collName, doc);
		}

		// 修改数据
		String id = "621d94555dd3312fd4a6f4f0";
		Document newdoc = new Document();
		newdoc.put("name", "时候");
		MongoDBUtil.instance.updateById(coll, id, newdoc);

		// 统计表
		System.out.println("行数 --> " + MongoDBUtil.instance.getCount(coll));
		System.out.println("条件查询、取第一个 --> " + coll.find(Filters.eq("_id", "621d94555dd3312fd4a6f4f0")).first());

		// 查询多个
		for (Document document : coll.find(Filters.eq("school", "NEFU1"))) {
			System.out.println("查询多个 --> " + document.toString());
		}

		// 根据ID查询
		Document doc = MongoDBUtil.instance.getById(coll, id);
		System.out.println("根据id查询 --> " + doc);

		// 删除表
		// MongoDBUtil.instance.deleteTable(dbName, collName);

		// 删除数据库
		// MongoDBUtil.instance.dropDb("testdb");
	}

}